import sys
import numpy as np
from astropy.io import fits
sys.path.append('aspired')
import aspired
science_frame = aspired.ImageReduction('examples/lhs6328.list')
science_frame.reduce()
#science_frame.inspect()
#science_frame.savefits(overwrite=True)
standard_frame = aspired.ImageReduction('examples/hiltner102.list')
standard_frame.reduce()
#standard_frame.inspect()
# Example data from SPRAT
# LHS6328 companion WD
science_data = science_frame.fits_data.data
# Example data from SPRAT
# Hiltner102
standard_data = standard_frame.fits_data.data
# Set the spectral and spatial direction
Saxis = 1
Waxis = 0
# spec mask
spec_mask = np.arange(science_frame.fits_data.header['NAXIS1'])
spatial_mask = np.arange(40,science_frame.fits_data.header['NAXIS2'])
# initialise the two aspired.TwoDSpec()
lhs6328 = aspired.TwoDSpec(science_data, spatial_mask=spatial_mask, spec_mask=spec_mask, rn=10., cr=False, gain=1., seeing=1.2, display=True)
hilt102 = aspired.TwoDSpec(standard_data, spatial_mask=spatial_mask, spec_mask=spec_mask, rn=10., cr=False, gain=1., seeing=1.2, display=True)
# automatically trace the spectrum
lhs6328.ap_trace(nsteps=20, recenter=False, prevtrace=(0, ), fittype='spline', bigbox=8)
hilt102.ap_trace(nsteps=20, recenter=False, prevtrace=(0, ), fittype='spline', bigbox=8)